package _back_tracking

import org.junit.Assert
import org.junit.Test

/**
 * https://leetcode.cn/problems/target-sum/description/
 * https://blog.csdn.net/asdfasaa/article/details/136179842
 *
 * 494. 目标和
 *
 * ```
 * 给你一个非负整数数组 nums 和一个整数 target 。
 * 向数组中的每个整数前添加 '+' 或 '-' ，然后串联起所有整数，可以构造一个 表达式 ：
 * 例如，nums = [2, 1] ，可以在 2 之前添加 '+' ，在 1 之前添加 '-' ，然后串联起来得到表达式 "+2-1" 。
 * 返回可以通过上述方法构造的、运算结果等于 target 的不同 表达式 的数目。
 *
 *
 * 示例 1：
 * 输入：nums = [1,1,1,1,1], target = 3
 * 输出：5
 * 解释：一共有 5 种方法让最终目标和为 3 。
 * -1 + 1 + 1 + 1 + 1 = 3
 * +1 - 1 + 1 + 1 + 1 = 3
 * +1 + 1 - 1 + 1 + 1 = 3
 * +1 + 1 + 1 - 1 + 1 = 3
 * +1 + 1 + 1 + 1 - 1 = 3
 *
 * 示例 2：
 * 输入：nums = [1], target = 1
 * 输出：1
 *
 *
 * 提示：
 * 1 <= nums.length <= 20
 * 0 <= nums[i] <= 1000
 * 0 <= sum(nums[i]) <= 1000
 * -1000 <= target <= 1000
 * ```
 *
 */
class leetcode_494 {
    // https://blog.csdn.net/asdfasaa/article/details/136179842

    @Test
    fun test_1() {
        val result = findTargetSumWays(nums = intArrayOf(1, 1, 1, 1, 1), target = 3)
        val expected = 5
        Assert.assertEquals(expected, result)
    }


    @Test
    fun test_2() {
        val result = findTargetSumWays(nums = intArrayOf(1), target = 1)
        val expected = 1
        Assert.assertEquals(expected, result)
    }

    private fun findTargetSumWays(nums: IntArray, target: Int): Int {
        /**
        回溯法：每次循环时，要尝试+和-
         */
        collect(nums, target, 0, 0, "")
        return count
    }

    private var count: Int = 0

    // path 用来打印数据
    private fun collect(nums: IntArray, target: Int, sum: Int, index: Int, path: String) {
        // 1 确认函数参数以及返回值

        // 2 确认终止条件
        println(path)
        if (index == nums.size) { // 说明移动到了数组后面
            //  收集结果
            if (target == sum) {
                count++
            }
            return
        }

        // 2. 递归步骤: 对当前数字 nums[index] 进行两种选择
        // 选择 '+'
        // 累加当前数字 (currentSum + nums[index])
        // 移动到下一个数字 (index + 1)
        collect(nums, target, sum + nums[index], index + 1, path + "+ ${nums[index]}")
        // 选择 '-'
        // 减去当前数字 (currentSum - nums[index])
        // 移动到下一个数字 (index + 1)
        collect(nums, target, sum - nums[index], index + 1, path + "+ ${nums[index]}")
    }
}